#pragma once
#include "google/protobuf/service.h"
#include <string>
#include <google/protobuf/descriptor.h>
#include <muduo/net/TcpServer.h>
#include <muduo/net/EventLoop.h>
#include <muduo/net/InetAddress.h>
#include <muduo/net/TcpConnection.h>
#include <unordered_map>

// 框架提供的专门发布rpc的网络对象
class RpcProvider
{
public:
    // 框架提供给外部服务，可以发布rpc的方法接口
    void NotifyService(::google::protobuf::Service *service);
    // 启动rpc服务，开启rpc远程网络服务调用
    void Run();

private:
    muduo::net::EventLoop m_evetloop;
    void OnConnection(const muduo::net::TcpConnectionPtr &conn);
    void OnMessage(const muduo::net::TcpConnectionPtr &conn, muduo::net::Buffer *buffer, muduo::Timestamp timestamp);
    // Closure的回调操作，用于序列化rpc的响应和网络发送
    void SendRpcResponse(const muduo::net::TcpConnectionPtr &conn, google::protobuf::Message *response);

    struct ServiceInfo
    {
        google::protobuf::Service *m_service; // save service object
        std::unordered_map<std::string, const google::protobuf::MethodDescriptor *> m_methodMap;
    };
    std::unordered_map<std::string, ServiceInfo> m_serviceMap;
};
